package com.auntvt.timo.jpax.core.service.impl;

import com.auntvt.timo.jpax.core.service.JpaService;
import com.auntvt.timo.jpax.core.utils.RepositoryUtil;
import org.springframework.data.domain.Example;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.Pageable;
import org.springframework.data.domain.Sort;
import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.transaction.annotation.Transactional;

import java.util.List;
import java.util.Optional;

/**
 * JpaService接口实现，对应JpaRepository接口的方法
 *
 * @author 小懒虫 <auntvt@163.com>
 * @date 2021/5/26
 */
public class JpaServiceImpl<T, ID> implements JpaService<T, ID> {

    /**
     * 对应实体类的持久层Repository
     */
    protected JpaRepository<T, ID> repository;

    /**
     * 获取对应实体类的持久层Repository接口
     *
     * @return Repository接口
     */
    @SuppressWarnings("unchecked")
    public JpaRepository<T, ID> getRepository() {
        if (repository == null) {
            repository = RepositoryUtil.getMatchRepository(JpaRepository.class, this.getClass());
        }
        return repository;
    }

    /**
     * 根据主键ID查询获取数据
     *
     * @param id 主键ID
     * @return 实体类对象
     */
    @Override
    public T getById(ID id) {
        return getRepository().findById(id).orElse(null);
    }

    /**
     * 根据主键ID查询获取数据
     *
     * @param id 主键ID
     * @return Optional对象
     */
    @Override
    public Optional<T> getOptionalById(ID id) {
        return getRepository().findById(id);
    }

    /**
     * 根据查询实例条件获取数据
     *
     * @param example 查询实例
     * @return 实体类对象
     */
    @Override
    public T get(Example<T> example) {
        return getRepository().findOne(example).orElse(null);
    }

    /**
     * 根据查询实例条件获取数据
     *
     * @param example 查询实例
     * @return Optional对象
     */
    @Override
    public Optional<T> getOptional(Example<T> example) {
        return getRepository().findOne(example);
    }

    /**
     * 获取全部数据
     *
     * @return 列表数据
     */
    @Override
    public List<T> lists() {
        return getRepository().findAll();
    }

    /**
     * 获取全部数据，并且进行排序
     *
     * @param sort 排序对象
     * @return 列表数据
     */
    @Override
    public List<T> lists(Sort sort) {
        return getRepository().findAll(sort);
    }

    /**
     * 根据查询实例条件获取数据
     *
     * @param example 查询实例
     * @return 列表数据
     */
    @Override
    public List<T> lists(Example<T> example) {
        return getRepository().findAll(example);
    }

    /**
     * 根据查询实例条件获取数据，并且进行排序
     *
     * @param example 查询实例
     * @param sort    排序对象
     * @return 列表数据
     */
    @Override
    public List<T> lists(Example<T> example, Sort sort) {
        return getRepository().findAll(example, sort);
    }

    /**
     * 根据主键ID列表获取对应的数据
     *
     * @param ids 主键ID列表
     * @return 列表数据
     */
    @Override
    public List<T> listById(Iterable<ID> ids) {
        return getRepository().findAllById(ids);
    }

    /**
     * 根据查询实例条件获取数据，并且进行分页
     *
     * @param pageable 分页配置对象
     * @return 分页数据
     */
    @Override
    public Page<T> pages(Pageable pageable) {
        return getRepository().findAll(pageable);
    }

    /**
     * 根据查询实例条件获取数据，并且进行分页
     *
     * @param example  查询实例
     * @param pageable 分页配置对象
     * @return 分页数据
     */
    @Override
    public Page<T> pages(Example<T> example, Pageable pageable) {
        return getRepository().findAll(example, pageable);
    }

    /**
     * 统计个数
     *
     * @return 数量
     */
    @Override
    public long count() {
        return getRepository().count();
    }

    /**
     * 根据查询实例条件统计个数
     *
     * @param example 查询实例
     * @return 数量
     */
    @Override
    public long count(Example<T> example) {
        return getRepository().count();
    }

    /**
     * 根据查询实例条件判断数据是否存在
     *
     * @param example 查询实例
     * @return Boolean
     */
    @Override
    public boolean exists(Example<T> example) {
        return getRepository().exists(example);
    }

    /**
     * 根据主键ID判断数据是否存在
     *
     * @param id 主键ID
     * @return Boolean
     */
    @Override
    public boolean existsById(ID id) {
        return getRepository().existsById(id);
    }

    /**
     * 保存数据
     *
     * @param entity 实体类对象
     * @return 保存后的数据
     */
    @Override
    public T save(T entity) {
        return getRepository().save(entity);
    }

    /**
     * 保存列表数据
     *
     * @param entities 列表数据
     * @return 保存后的数据
     */
    @Override
    @Transactional(rollbackFor = RuntimeException.class)
    public List<T> saveAll(Iterable<T> entities) {
        return getRepository().saveAll(entities);
    }

    /**
     * 立即保存数据
     *
     * @param entity 实体类对象
     * @return 保存后的数据
     */
    @Override
    public T saveAndFlush(T entity) {
        return getRepository().saveAndFlush(entity);
    }

    /**
     * 立即保存列表数据
     *
     * @param entities 实体类列表
     * @return 保存后的数据
     */
    @Override
    public List<T> saveAllAndFlush(Iterable<T> entities) {
        return getRepository().saveAllAndFlush(entities);
    }

    /**
     * 根据主键ID删除数据
     *
     * @param id 主键ID
     */
    @Override
    public void deleteById(ID id) {
        getRepository().deleteById(id);
    }

    /**
     * 删除对应实体类的数据
     *
     * @param entity 实体类对象
     */
    @Override
    public void delete(T entity) {
        getRepository().delete(entity);
    }

    /**
     * 删除全部数据
     */
    @Override
    @Transactional(rollbackFor = RuntimeException.class)
    public void deleteAll() {
        getRepository().deleteAll();
    }

    /**
     * 删除对应实体类列表的数据
     *
     * @param entities 数据列表
     */
    @Override
    @Transactional(rollbackFor = RuntimeException.class)
    public void deleteAll(Iterable<? extends T> entities) {
        getRepository().deleteAll(entities);
    }

    /**
     * 删除对应主键ID列表的数据
     *
     * @param ids 主键ID列表
     */
    @Override
    public void deleteAllById(Iterable<? extends ID> ids) {
        getRepository().deleteAllById(ids);
    }

    /**
     * 批量删除对应主键ID列表的数据
     *
     * @param ids 主键ID列表
     */
    @Override
    public void deleteAllByIdInBatch(Iterable<ID> ids) {
        getRepository().deleteAllByIdInBatch(ids);
    }

    /**
     * 批量删除对应实体类列表的数据
     *
     * @param entities 数据列表
     */
    @Override
    public void deleteAllInBatch(Iterable<T> entities) {
        getRepository().deleteAllInBatch(entities);
    }

    /**
     * 批量删除全部数据
     */
    @Override
    @Transactional(rollbackFor = RuntimeException.class)
    public void deleteAllInBatch() {
        getRepository().deleteAllInBatch();
    }
}
